#!/usr/bin/env python
# -*- coding: UTF-8 -*-
'''
项目名称 ：homework1017 
脚本名称 ：rewrite_xueqiu.py
开发环境 ：PyCharm 
开发人员 ：林
编写时间 ：2021/10/19 20:47 
'''
import json
from mitmproxy import http

def response(flow:http.HTTPFlow):
    """
    将雪球-行情列表中前三只股票涨跌幅修改，实现股票颜色的变换测试
    :param flow:
    :return:
    """
    #匹配请求url
    if "quote.json" in flow.request.pretty_url :
        #将响应报文转换为json格式
        data = json.loads(flow.response.content)
        #修改响应报文中的数据
        data['data']['items'][0]['quote']['name'] = 'hogwarts001'
        data['data']['items'][0]['quote']['percent'] = 0.01
        data['data']['items'][1]['quote']['name'] = 'hogwarts002'
        data['data']['items'][1]['quote']['percent'] = 0.00
        data['data']['items'][2]['quote']['name'] = 'hogwarts002'
        data['data']['items'][2]['quote']['percent'] = -0.01
        #将修改后的数据重新写入响应报文
        flow.response.text = json.dumps(data)